package hxy.bytecode.lambda;

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import lombok.Data;
import org.junit.Test;

/**
 * Created by Jackielee on 2017
 *
 * @author: lizhilong
 * @date: 2017-11-24 10:13:57
 * @Copyright: 2017 www.aliyun.com Inc. All rights reserved.
 */
public class StreamListToMap {
    List<Student> list = Arrays.asList(new Student(1, 18, "阿龙"),
            new Student(2, 17, "小花"),
            new Student(3, 17, "阿浪"));

    /**
     * 将list转成map
     */
    @Test
    public void listToMapByObjectValue() {
        // value 为对象 student -> student jdk1.8返回当前对象
        Map<Integer, Student> map = list.stream().collect(Collectors.toMap(Student::getId, student -> student));
        // 遍历打印结果
        map.forEach((key, value) -> {
            System.out.println("key: " + key + "    value: " + value);
        });
    }
    //value为对象执行结果：
//
//key: 1    value: Student [id=1, age=18, name=阿龙, gender=0]
//key: 2    value: Student [id=2, age=17, name=小花, gender=1]
//key: 3    value: Student [id=3, age=17, name=阿浪, gender=2]
    /**
     * 取list的值作为key或者vlaue
     */
    @Test
    public void listToMapByNameValue() {
        // value 为对象中的属性
        Map<Integer, String> map = list.stream().collect(Collectors.toMap(Student::getId, Student::getName));
        map.forEach((key, value) -> {
            System.out.println("key: " + key + "    value: " + value);
        });
    }

//value为字段执行结果：
//
//key: 1    value: 阿龙
//key: 2    value: 小花
//key: 3    value: 阿浪
//要注意的是map的key必须唯一，所以有可能出现不唯一的时候，就会报错
    @Test
    public void listToMapByAgeKey() {
        // value 为对象中的属性
        Map<Integer, String> map = list.stream().collect(Collectors.toMap(Student::getAge, Student::getName));
    }

    //执行结果：
//
//java.lang.IllegalStateException: Duplicate key 小花
//因为age有相同就会报错
//此时stream包下Collectors.toMap方法有一个重载方法的参数，这个参数可以传一个合并的函数解决冲突
    @Test
    public void listToMapByAgeKey1() {
        // value 为对象中的属性
        Map<Integer, String> map = list.stream().collect(
                Collectors.toMap(Student::getAge, Student::getName, (key1, key2) -> key1)
        );
        map.forEach((key, value) -> {
            System.out.println("key: " + key + "    value: " + value);
        });
    }
}

//
//执行结果：
//key: 17    value: 小花
//key: 18    value: 阿龙

@Data
class Student {
    int id;
    int age;
    String name;

    Student(int id, int age, String name) {
        this.id = id;
        this.age = age;
        this.name = name;
    }
}


//————————————————
//版权声明：本文为CSDN博主「熬夜是小狗」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。
//原文链接：https://blog.csdn.net/icannotdebug/article/details/78621705